import os
from dotenv import load_dotenv

load_dotenv()

class Settings:
    LLM_URL = os.getenv("LLM_URL", "http://172.20.0.89:8011/v1")
    LLM_NAME = os.getenv("LLM_NAME", "qwen3-30b3a")
    REDIS_HOST = os.getenv("REDIS_HOST", "172.20.0.89")
    REDIS_PORT = int(os.getenv("REDIS_PORT", 6379))
    MILVUS_HOST = os.getenv("MILVUS_HOST", "172.20.0.89")
    MILVUS_PORT = os.getenv("MILVUS_PORT", "19530")
    MILVUS_COLLECTION_NAME = os.getenv("MILVUS_COLLECTION_NAME", "Policies")
    EMBEDDING_URL = os.getenv("EMBEDDING_URL", "http://172.20.0.89:11434")
    EMBEDDING_MODEL = os.getenv("EMBEDDING_MODEL", "bge-m3:latest")
    RETRIEVE_NUM = int(os.getenv("RETRIEVE_NUM", 5))
    OPENAI_API_KEY = os.getenv("API_KEY", "")